查看原文
其他

正确规避常见的 Go 并发陷阱

仓屋 aoho求索 2022-09-08

作为一门并发性能优越的语言,Go虽然降低了协程并发开发的难度,但也存在一些「并发陷阱」,这就需要我们在开发时额外注意。

我们再来介绍与Go 并发相关的几个小技巧,帮助你规避 Go 并发开发的一些陷阱。

闭包传递参数的问题

首先是循环并发时闭包传递参数的问题,如下错误例子所示:

func main()  {
for i := 0 ; i < 5 ; i++{
go func() {
fmt.Println("current i is ", i)
}()
}
time.Sleep(time.Second)
}

这段代码极有可能的输出为:

current i is 5
current i is 5
current i is 5
current i is 5
current i is 5

这是因为 i 使用的地址空间在循环中被复用,在 goroutine 执行时,i 的值可能在被主 goroutine 修改,而此时其他 goroutine 也在读取使用,从而导致了并发错误。针对这种错误可以通过「复制拷贝」或者「传参拷贝」的方式规避,如下所示:

func main() {
for i := 0 ; i < 5 ; i++{
go func(v int) {
fmt.Println("current i is", v)
}(i)
}
time.Sleep(time.Second)
}

panic 异常

上一篇介绍 panic 时我们了解到 panic 异常的出现会导致 Go 程序的崩溃。但其实即使 panic 是出现在其他启动的子 goroutine 中,也会导致 Go 程序的崩溃退出,同时 panic 只能捕获 goroutine 自身的异常,因此「对于每个启动的 goroutine,都需要在入口处捕获 panic,并尝试打印堆栈信息并进行异常处理,从而避免子 goroutine 的 panic 导致整个程序的崩溃退出」。如下面的例子所示:

func RecoverPanic() {
    // 从 panic 中恢复并打印栈信息
if e := recover(); e != nil {
buf := make([]byte1024)
buf = buf[:runtime.Stack(buf, false)]
fmt.Printf("[PANIC]%v\n%s\n", e, buf)
}
}
func main() {
for i:= 0 ; i < 5 ; i++{
go func() {
            // defer 注册 panic 捕获函数
defer RecoverPanic()
dothing()
}()
}
}

超时控制

最后一个技巧是要「善于结合使用 select、timer 和 context 进行超时控制」。在 goroutine 中进行一些耗时较长的操作,最好都加上超时timer,在并发的时候也要传递 context,这样在取消的时候就不会有遗漏,进而达到回收 goroutine 的目的,避免内存泄漏的发生。如下面的例子所示,通过 select 同时监听任务和定时器状态,在定时器到达而任务未完成之时,提前结束任务,清理资源并返回。

select {
// do logic process
case msg <- input:
   ....
// has been canceled
case <-ctx.Done():
    // ...资源清理
    return
// 2 second timeout    
case <-time.After(time.Second * 2)  
    // ...资源清理
    return
default:
}

小结

本文我们主要介绍了Go 中常见的一些并发开发技巧。笔者也只是抛砖引玉,提出了并发开发技巧的三个技巧,其实不仅仅是 Go 并发,大家在日常开发中需要注意总结各个方面的开发技巧。

推荐阅读


微服务架构中使用 ELK 进行日志采集以及统一处理

没有 try-catch,该如何处理 Go 错误异常?

订阅最新文章,欢迎关注我的公众号


您可能也对以下帖子感兴趣

文章有问题?点此查看未经处理的缓存